from flask import g
from functools import wraps


def login_required(func):
    """强制登录"""
    @wraps(func)
    def wrapper(*args, **kwargs):
        # 在视图函数之情前判断是否登录
        # print('uid--->', g.user_id)
        if g.user_id:
            func(*args, **kwargs)
        else:
            # 没有登录
            return {'message': '请先登录'}, 401
    return wrapper
